#pragma once

#include <iostream>
#include <string>
#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <signal.h>
#include "Log.hpp"
#include "Common.hpp"

using namespace LogModule;
std::string dev = "/dev/null";
// 将业务进行守护进程化

// if nochdir == 0, 工作路径改为 "/"
// if noclose == 0, 标准输入，标准输出，标准错误重定向到 /dev/null
void Daemon(int nochdir, int noclose)
{
    // 1. 忽略 IO, 子进程退出等相关信号
    signal(SIGPIPE, SIG_IGN);
    signal(SIGCHLD, SIG_IGN);

    // 2. 父进程直接退出
    if (fork() > 0)
        exit(0);

    // 3. 只有子进程，子进程成为孤儿进程，并守护进程化
    setsid();

    if (nochdir == 0)
        chdir("/");

    if (noclose == 0)
    {
        int fd = ::open(dev.c_str(), O_RDWR);
        if (fd < 0)
        {
            LOG(LogLevel::FATAL) << "open " << dev << "failed!";
            exit(OPEN_ERR);
        }
        else
        {
            dup2(fd, 0);
            dup2(fd, 1);
            dup2(fd, 2);
            close(fd);
        }
    }
}